class TrafficModelParam:
    traffic_dim: int
    noise_dim: int

    def __init__(self, traffic_dim: int, noise_dim: int):
        """
        Traffic模型 参数
        :param traffic_dim: 流量维度
        :param noise_dim: 噪声维度
        """
        self.traffic_dim = traffic_dim
        self.noise_dim = noise_dim
